"""
1235. 规划兼职工作
困难
相关标签
premium lock icon
相关企业
提示
你打算利用空闲时间来做兼职工作赚些零花钱。

这里有 n 份兼职工作，每份工作预计从 startTime[i] 开始到 endTime[i] 结束，报酬为 profit[i]。

给你一份兼职工作表，包含开始时间 startTime，结束时间 endTime 和预计报酬 profit 三个数组，请你计算并返回可以获得的最大报酬。

注意，时间上出现重叠的 2 份工作不能同时进行。

如果你选择的工作在时间 X 结束，那么你可以立刻进行在时间 X 开始的下一份工作。



示例 1：



输入：startTime = [1,2,3,3], endTime = [3,4,5,6], profit = [50,10,40,70]
输出：120
解释：
我们选出第 1 份和第 4 份工作，
时间范围是 [1-3]+[3-6]，共获得报酬 120 = 50 + 70。
示例 2：



输入：startTime = [1,2,3,4,6], endTime = [3,5,10,6,9], profit = [20,20,100,70,60]
输出：150
解释：
我们选择第 1，4，5 份工作。
共获得报酬 150 = 20 + 70 + 60。
示例 3：



输入：startTime = [1,1,1], endTime = [2,3,4], profit = [5,6,4]
输出：6


提示：

1 <= startTime.length == endTime.length == profit.length <= 5 * 10^4
1 <= startTime[i] < endTime[i] <= 10^9
1 <= profit[i] <= 10^4
"""
from bisect import bisect_left
from typing import List


class Solution:

    def jobScheduling(self, startTime: List[int], endTime: List[int], profit: List[int]) -> int:
        """
        题目理解有误，未考虑k 这题是1235d的解法
        dp,根据结束时间，考虑选当前和不选当前的情况,
        定义数组三维数组cache[max(left)][i][2],以right作为索引，存储个二维数组，二维数组每一项为[left,价值]
        定义数组dp[],计算每项的最最优情况，条件表达式为
        对于cache[i],执行 max(dp[i-1],dp[cache[i][j1][0] - 1] + cache[i][j1][1],...dp[cache[i][jn][0] - 1] + cache[i][jn][1])
        同一天内有多个则同时计算max
        :param events:
        :param k:
        :return:
        """
        jobs = zip(startTime, endTime, profit)
        print(jobs)
        print(type(jobs))
        jobs = list(zip(startTime, endTime, profit))
        print(jobs)
        print(type(jobs))
        jobs = [list(x) for x in zip(startTime, endTime, profit)]
        print(jobs)
        print(type(jobs))

        jobs = [list(x) for x in zip(startTime, endTime, profit)]

        return self.maxValue_test1(jobs)
    def maxValue_test1(self, events: List[List[int]]) -> int:
        """
        题目理解有误，未考虑k 这题是1235d的解法
        dp,根据结束时间，考虑选当前和不选当前的情况,
        定义数组三维数组cache[max(left)][i][2],以right作为索引，存储个二维数组，二维数组每一项为[left,价值]
        定义数组dp[],计算每项的最最优情况，条件表达式为
        对于cache[i],执行 max(dp[i-1],dp[cache[i][j1][0] - 1] + cache[i][j1][1],...dp[cache[i][jn][0] - 1] + cache[i][jn][1])
        同一天内有多个则同时计算max
        :param events:
        :param k:
        :return:
        """
        #右边界最大值
        max1 = max(e[1] for e in events )
        #先定义二维
        cache = [[] for _ in range(max1 + 1)]  # 每个元素是一个 list of [start, value]
        #二维每项是个数组
        for start, end, value in events:
            cache[end].append([start, value])

        #内存优化?
        dp = [0] * (max1+1)
        dp[0] = 0
        #状态转移
        for i in range(1,max1+1):
            if(cache[i] == []):
                dp[i] = dp[i-1]
            else:
                maxValue = dp[i - 1]
                list = cache[i]
                #i是left，j
                for left,value in list:
                    maxValue = max(maxValue,dp[left] + value) #边界可以相等，不需要减一
                    # maxValue = max(maxValue,dp[left - 1] + value)
                dp[i] = maxValue
        return dp[max1]
    def jobScheduling_res1(self, startTime: List[int], endTime: List[int], profit: List[int]) -> int:
        jobs = sorted(zip(endTime, startTime, profit))  # 按照结束时间排序
        f = [0] * (len(jobs) + 1)
        for i, (_, st, p) in enumerate(jobs):
            j = bisect_left(jobs, (st+1 ,), hi=i)  # hi=i 表示二分上界为 i（默认为 n） todo 为什么st+1是对的
            # hi=i 表示二分上界为 i（默认为 n）
            # 状态转移中，为什么是 j 不是 j+1：上面算的是 > st，-1 后得到 <= st，但由于还要 +1，抵消了
            f[i+1] = max(f[i], f[j] + p)
        return f[-1]


if __name__ == '__main__':
    print(Solution().jobScheduling_res1(startTime = [1,2,3,3], endTime = [3,4,5,6], profit = [50,10,40,70])) #120
    # print(Solution().jobScheduling_res1(startTime=[1,2,3,4,6], endTime = [3,5,10,6,9], profit = [20,20,100,70,60])) #150
    # print(Solution().jobScheduling_res1(startTime = [1,1,1], endTime = [2,3,4], profit = [5,6,4])) #6

    # jobs = [
    #     (3, 1, 50),
    #     (4, 2, 10),
    #     (5, 3, 40),  # 当前处理这个工作
    #     (6, 3, 70)
    # ]
    # i = len(jobs)
    # j = bisect_left(jobs, (4,), hi=i)
    # print(j)#2
    # j = bisect_left(jobs, (3,), hi=i)
    # print(j)#1
